﻿using System;
using System.Diagnostics;
using System.Globalization;
using System.Runtime.InteropServices;
using System.ComponentModel.Design;
using Microsoft.Win32;
using Microsoft.VisualStudio;
using Microsoft.VisualStudio.Shell.Interop;
using Microsoft.VisualStudio.OLE.Interop;
using Microsoft.VisualStudio.Shell;
using EnvDTE;
using EnvDTE80;
using System.IO;
using QuickPasteItLib;
using System.Windows.Forms;

namespace QuickPasteIt.QuickPasteItVs
{
    /// <summary>
    /// This is the class that implements the package exposed by this assembly.
    ///
    /// The minimum requirement for a class to be considered a valid package for Visual Studio
    /// is to implement the IVsPackage interface and register itself with the shell.
    /// This package uses the helper classes defined inside the Managed Package Framework (MPF)
    /// to do it: it derives from the Package class that provides the implementation of the 
    /// IVsPackage interface and uses the registration attributes defined in the framework to 
    /// register itself and its components with the shell.
    /// </summary>
    // This attribute tells the PkgDef creation utility (CreatePkgDef.exe) that this class is
    // a package.
    [PackageRegistration(UseManagedResourcesOnly = true)]
    // This attribute is used to register the information needed to show this package
    // in the Help/About dialog of Visual Studio.
    [InstalledProductRegistration("#110", "#112", "1.0", IconResourceID = 400)]
    // This attribute is needed to let the shell know that this package exposes some menus.
    [ProvideMenuResource("Menus.ctmenu", 1)]
    [Guid(GuidList.guidQuickPasteItVsPkgString)]
    public sealed class QuickPasteItVsPackage : Package
    {
        /// <summary>
        /// Default constructor of the package.
        /// Inside this method you can place any initialization code that does not require 
        /// any Visual Studio service because at this point the package object is created but 
        /// not sited yet inside Visual Studio environment. The place to do all the other 
        /// initialization is the Initialize method.
        /// </summary>
        public QuickPasteItVsPackage()
        {
            Debug.WriteLine(string.Format(CultureInfo.CurrentCulture, "Entering constructor for: {0}", this.ToString()));
        }



        /////////////////////////////////////////////////////////////////////////////
        // Overridden Package Implementation
        #region Package Members

        /// <summary>
        /// Initialization of the package; this method is called right after the package is sited, so this is the place
        /// where you can put all the initialization code that rely on services provided by VisualStudio.
        /// </summary>
        protected override void Initialize()
        {
            Debug.WriteLine (string.Format(CultureInfo.CurrentCulture, "Entering Initialize() of: {0}", this.ToString()));
            base.Initialize();

            // Add our command handlers for menu (commands must exist in the .vsct file)
            OleMenuCommandService mcs = GetService(typeof(IMenuCommandService)) as OleMenuCommandService;
            if ( null != mcs )
            {
                // Create the command for the solution explorer menu item.
                CommandID menuCommandID = new CommandID(GuidList.guidQuickPasteItVsCmdSet, (int)PkgCmdIDList.cmdidQuickPasteIt);
                MenuCommand menuItem = new MenuCommand(SolExplorerMenuItemCallback, menuCommandID);
                mcs.AddCommand( menuItem );

                // Create the command for the code editor menu item.
                CommandID menuCommandIDEditor = new CommandID(GuidList.guidQuickPasteItVsCmdSetEditor, (int)PkgCmdIDList.cmdidQuickPasteItEdtitor);
                MenuCommand menuItemEditor = new MenuCommand(EditorMenuItemCallback, menuCommandIDEditor);
                mcs.AddCommand( menuItemEditor );
            }
        }
        #endregion

        /// <summary>
        /// This function is the callback used to execute a command when the a menu item is clicked.
        /// See the Initialize method to see how the menu item is associated to this function using
        /// the OleMenuCommandService service and the MenuCommand class.
        /// </summary>
        private void EditorMenuItemCallback(object sender, EventArgs e)
        {
            DTE applicationObject = GetService(typeof(SDTE)) as DTE;
            QuickPasteIt qpi = new QuickPasteIt();

            try
            {

                // Check if any code is selected
                if (applicationObject.ActiveDocument != null &&
                    applicationObject.ActiveDocument.Selection != null &&
                    ((TextDocument)(applicationObject.ActiveDocument.Object("TextDocument"))).Selection.Text != "")
                {
                    // Pastebin the selected code
                    FileInfo currDoc = new FileInfo(applicationObject.ActiveDocument.FullName);
                    PasteLang codeLang = FileTools.GetPasteLangFromExtension(currDoc.Extension);
                    qpi.PasteBin(((TextDocument)(applicationObject.ActiveDocument.Object("TextDocument"))).Selection.Text, codeLang);
                }

                // No code selected - paste the whole document if the current document exists
                else if(applicationObject.ActiveDocument != null &&
                       applicationObject.ActiveDocument.FullName != null &&
                       File.Exists(applicationObject.ActiveDocument.FullName))
                {
                    // Perform the pastebin
                    qpi.PasteBin(applicationObject.ActiveDocument.FullName);
                }
            }
            catch
            {
                ShowMessage("An error occurred while attempting to make the paste", true);
            }
        }

        /// <summary>
        /// This function is the callback used to execute a command when the a menu item is clicked.
        /// See the Initialize method to see how the menu item is associated to this function using
        /// the OleMenuCommandService service and the MenuCommand class.
        /// </summary>
        private void SolExplorerMenuItemCallback(object sender, EventArgs e)
        {
            DTE2 applicationObject = (DTE2)Package.GetGlobalService(typeof(SDTE));
            UIHierarchy solutionExplorer = applicationObject.ToolWindows.SolutionExplorer;

            var items = solutionExplorer.SelectedItems as Array;

            if (items == null || items.Length != 1)
            {
                ShowMessage("You can only pastebin one file at a time.", true);
                return;
            }

            string prjPath = "";

            foreach (UIHierarchyItem hitem in items)
            {
                ProjectItem prjItem = hitem.Object as ProjectItem;
                prjPath = prjItem.Properties.Item("FullPath").Value.ToString();
            }

            QuickPasteIt qpi = new QuickPasteIt();
            qpi.PasteBin(prjPath);
        }


        /// <summary>
        /// Displays a message to the user.
        /// </summary>
        private void ShowMessage(string message, bool error)
        {
            OLEMSGICON icon = OLEMSGICON.OLEMSGICON_INFO;
            
            if(error)
                icon = OLEMSGICON.OLEMSGICON_CRITICAL;

            IVsUIShell uiShell = (IVsUIShell)GetService(typeof(SVsUIShell));
            Guid clsid = Guid.Empty;
            int result;
            Microsoft.VisualStudio.ErrorHandler.ThrowOnFailure(uiShell.ShowMessageBox(
                       0,
                       ref clsid,
                       "",
                       message,
                       string.Empty,
                       0,
                       OLEMSGBUTTON.OLEMSGBUTTON_OK,
                       OLEMSGDEFBUTTON.OLEMSGDEFBUTTON_FIRST,
                       icon,
                       0,        // false
                       out result));
        }
    }
}
